# import numpy as np
# import matplotlib.pyplot as plt
# plt.plot(np.random.randn(50).cumsum())

nums = [i*i for i in range(1, 10)]

nums = [i*i for i in range(1, 10) if i > 3 else i > 3]


print(nums)